#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2021/6/1 23:15
# @Author  : huidong.bai
# @File    : demo.py
# @Software: PyCharm

import wave
import matplotlib.pyplot as pl
import numpy as np
# 打开WAV文档
f = wave.open(r"../record_audio/output.wav", "rb")
# 读取格式信息
# (nchannels, sampwidth, framerate, nframes, comptype, compname)
params = f.getparams()
nchannels, sampwidth, framerate, nframes = params[:4]
print(f'通道数:{nchannels}')
print(f'sampwidth:{sampwidth}')
print(f'采样率:{framerate}')
print(f'总帧数:{nframes}')
# 读取波形数据
str_data = f.readframes(nframes)
f.close()
#将波形数据转换为数组
wave_data = np.fromstring(str_data, dtype=np.short)
# for i in wave_data:
#     print(i)
wave_data.shape = -1, nchannels
wave_data = wave_data.T
# wave_data[abs(wave_data) < 200] = 0
wave_data[abs(wave_data) < 150] = 0
time = np.arange(0, nframes) * (1.0 / framerate)
# 绘制波形
pl.subplot(211)
pl.plot(time, wave_data[0])
pl.subplot(212)
pl.plot(time, wave_data[1], c="g")
pl.xlabel("time (seconds)")
pl.show()
